from rest_framework import serializers
from hospitals.models import DepartmentModel, HospitalModel


class DepartmentModelSer(serializers.ModelSerializer):
    class Meta:
        model = DepartmentModel
        fields = ['id', 'name', 'direction', 'cover']


class HospitalModelSer(serializers.ModelSerializer):
    class Meta:
        model = HospitalModel
        fields = ['id', 'name', 'cover', 'tel', 'get_level_display', 'profile', 'address']
